Skip to content

Conversation

@bgctw
Copy link
Member

@bgctw bgctw commented Apr 30, 2025

No description provided.

bgctw added 7 commits April 28, 2025 10:31
move n_batch from solver to problem

remove problem responsibilty of putting dataloader to gpu but do it in fitting and predicting functions

fix error in solver of properly updating problem.phi_unc
because exlementwise(exp) failed on AD on GPU
test on GPU
@bgctw bgctw merged commit c931205 into main Apr 30, 2025
6 checks passed
@bgctw bgctw deleted the dev branch April 30, 2025 15:01
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants